simple array sum hackerrank python